import os
import pandas as pd
import numpy as np
import shutil
from pdf2image import convert_from_path

os.makedirs('低pdfs', exist_ok=True)
os.makedirs('中pdfs', exist_ok=True)
os.makedirs('高pdfs', exist_ok=True)
# 读取 灾点文件记录表test.xls
def read_excel(file_path1,file_path2):
    # 读取Excel文件
    df1 = pd.read_excel(file_path1)
    df2 = pd.read_excel(file_path2)

    # 将Excel文件中的数据转换为字典
    data1 = df1.to_dict(orient='records')
    data2 = df2.to_dict(orient='records')

    # # 返回字典
    # return data

    # 读取字典数据中的第二列并打印出来
    # for item in data1:
    #     print(item['UID'])
    # for item in data2:
    #     print(item['UID'])
    #     print(item['PUID'])

    # 比较data1 中的UID是否在data2的PUID中，如果相同则获取data2中的数据并打印内容
    for item1 in data1:
        for item2 in data2:
            if item1['UID'] == item2['PUID']:
                #判断item2['风险评估']是否等于 中
                if item1['风险评估'] == '中':
                    # 判断受否为pdf文件
                    if item2['名称'].endswith('.pdf'):
                        # print(item1['UID'],item2['PUID'],'相同')
                        print(item2['路径'] + item2['名称'])
                        pdf_file = 'E:\\zdf\\备份\\430321湖南省湘潭市湘潭县\\'+item2['路径'] + item2['名称']
                        pdf_new_name = str(item2['统一编号']) + '-' + item2['名称'] + '.pdf'
                        # 将路径pdf_file路径中的文件存储在根目录下的pdfs文件夹中,并重命名为pdf_new_name
                        shutil.copy(pdf_file, '中pdfs')
                        os.rename('中pdfs\\' + item2['名称'], '中pdfs\\' + pdf_new_name)
                    else:
                        pass
                elif item1['风险评估'] == '低':
                     # 判断受否为pdf文件
                    if item2['名称'].endswith('.pdf'):
                        # print(item1['UID'],item2['PUID'],'相同')
                        print(item2['路径'] + item2['名称'])
                        pdf_file = 'E:\\zdf\\备份\\430321湖南省湘潭市湘潭县\\'+item2['路径'] + item2['名称']
                        pdf_new_name = str(item2['统一编号'])+ '-' + item2['名称'] + '.pdf'
                        # 将路径pdf_file路径中的文件存储在根目录下的pdfs文件夹中,并重命名为pdf_new_name
                        shutil.copy(pdf_file, '低pdfs')
                        os.rename('低pdfs\\' + item2['名称'], '低pdfs\\' + pdf_new_name)
                    else:
                        pass
                elif item1['风险评估'] == '高':
                     # 判断受否为pdf文件
                    if item2['名称'].endswith('.pdf'):
                        # print(item1['UID'],item2['PUID'],'相同')
                        print(item2['路径'] + item2['名称'])
                        pdf_file = 'E:\\zdf\\备份\\430321湖南省湘潭市湘潭县\\'+item2['路径'] + item2['名称']
                        pdf_new_name = str(item2['统一编号'])+ '-' + item2['名称'] + '.pdf'
                        # 将路径pdf_file路径中的文件存储在根目录下的pdfs文件夹中,并重命名为pdf_new_name
                        shutil.copy(pdf_file, '高pdfs')
                        os.rename('高pdfs\\' + item2['名称'], '高pdfs\\' + pdf_new_name)
                    else:
                        pass
                else:
                    pass
            else:
                pass

read_excel('切坡建房主表20240913.xls','灾点文件记录表20240913.xls' )
